package com.itany.shop.util;




import com.itany.shop.exception.DataAccessException;
import org.apache.commons.dbcp.BasicDataSourceFactory;

import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Properties;

/**
 * @author MissChen
 * @version 1.0
 * @date 2023 年12月20日 13:34
 */
public class JDBCUtil {
    private static DataSource ds;
    private static ThreadLocal<Connection> local;

    static {
        local = new ThreadLocal<>();
        Properties p = new Properties();
        try {
            p.load(JDBCUtil.class.getClassLoader().getResourceAsStream("dataSource.properties"));
            ds = BasicDataSourceFactory.createDataSource(p);

        } catch (IOException e) {
            e.printStackTrace();
            System.out.println("JDBCUtil初始化失败");
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("JDBCUtil初始化失败");
        }

    }

    /**
     * 使用数据库连接池获取一个线程单例的Connection对象
     * @return
     */
    public static Connection getConnection(){

        //从线程池中获取Connection对象
        Connection conn=local.get();

        //如果没有获取到链接，表示连接尚未存在
        if(conn==null){
            try {
                //从连接池中获取一个连接
                conn=ds.getConnection();

                //当拿到连接之后，将连接放入线程池中
                //当下次从线程池中获取的时候
                local.set(conn);


            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("获取连接失败");
            }
        }
        return conn;
    }

    /**
     * 释放指定的资源
     * @param conn
     * @param ps
     * @param rs
     */
    public static void close(Connection conn, PreparedStatement ps, ResultSet rs){
        try {
            if (conn!=null){
                conn.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放连接失败");
        }
        try {
            if (ps!=null){
                ps.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放状态集失败");
        }
        try {
            if (rs!=null){
                rs.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放连结果集失败");
        }
    }

    /**
     * 释放当前线程中的相关的连接
     * 一般用于业务层事务执行完毕后释放
     */
    public static void close(){

        //从线程池中取出当前的连接，如果没有取到连接，说明已经释放了，无需做操作
        Connection conn=local.get();
        try {
            //如果获取到了连接，说明尚未释放
            if(conn!=null){

                //则需要释放线程池中的连接
                conn.close();

                //删除
                local.remove();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放连接失败");
        }
    }


}
